Partial function in Scala